package com.xiaoxin_project.netty;

import io.netty.channel.ChannelInitializer;
import io.netty.channel.ChannelPipeline;
import io.netty.channel.socket.SocketChannel;
import io.netty.handler.codec.http.HttpObjectAggregator;
import io.netty.handler.codec.http.HttpServerCodec;
import io.netty.handler.codec.http.websocketx.WebSocketServerProtocolHandler;
import io.netty.handler.stream.ChunkedWriteHandler;
import io.netty.handler.timeout.IdleStateHandler;
import org.springframework.stereotype.Component;

@Component
public class WSServerInitializer extends ChannelInitializer<SocketChannel> {

    protected void initChannel(SocketChannel channel) throws Exception {
        ChannelPipeline pipeline = channel.pipeline();
        //Websocket基于http协议，所以要有http编解码器
        pipeline.addLast(new HttpServerCodec());
        //对写大数据流的支持
        pipeline.addLast(new ChunkedWriteHandler());
        //对httpMessage进行聚合，聚合成FullHttpRequest或FullHttpResponse
        //几乎都会使用到此handler
        pipeline.addLast(new HttpObjectAggregator(1024 * 64));

        //========================= 以上用于支持http协议 ==========================

        //========================= 增加心跳支持start  ===========================
        //针对客户端，如果在1分钟时没有向服务端发送读写心跳(ALL)则主动断开,如果是读空闲或写空闲不作任何处理
        pipeline.addLast(new IdleStateHandler(8,10,12));
        //自定义的空闲状态检测
        pipeline.addLast(new HeartBeatHandler());

        //========================= 增加心跳支持end ==============================




        /**
         *  websocket服务器处理的协议，用于给指定的客户端连接访问的路由 ： /ws
         * 帮忙处理握手动作：handshaking（close，ping，pong） =  心跳机制
         * 对于websocket来说，都是以frames进行传输的，不同类型的数据对应不同类型的frames
         **/
        pipeline.addLast(new WebSocketServerProtocolHandler("/ws"));

        //自定义handler
        pipeline.addLast(new com.xiaoxin_project.netty.ChatHandler());
    }
}
